Comparison of operation functions of PyTorch tensor and NumPy array